import pandas as pd
from sqlalchemy import create_engine

# 电脑中的文件路径默认使用\， 这个时候需要在路径前面加一个r（ 转义符） 避免路径里面的\被转义。


# 一、读取Excel
pd_read_excel = pd.read_excel("file\\report.xlsx", sheet_name=0, index_col=0, header=1)
print(f'pd_read_excel:\n{pd_read_excel},type:', type(pd_read_excel))

# 二、读取csv文件
# 如果文件是逗号分隔，则可以不写
pd_read_csv = pd.read_csv("file\\report.csv", sep='\t')
print(f'pd_read_csv:\n{pd_read_csv},type:', type(pd_read_csv))

# 三、读取txt文件
# 必须要写分隔符
pd_read_txt = pd.read_table("file\\report.csv", sep='\t')
print(f'pd_read_txt:\n{pd_read_txt},type:', type(pd_read_txt))


# 四、读取Mysql数据库数据
def read_mysql_data():
    # 使用engine的方式，不会有警告
    engine = create_engine('mysql+pymysql://%s:%s@%s:%s/%s?charset=utf8'
                           % ('root', '123456', '127.0.0.1', '3306', 'geek'))
    sql = "select * from employee"
    pd_read_sql = pd.read_sql(sql, engine)
    return pd_read_sql


pd_read_mysql = read_mysql_data()
print(f'pd_read_mysql:\n{pd_read_mysql},type:', type(pd_read_mysql))
print('查看行信息：pd_read_mysql.info()', pd_read_mysql.info())

